from tensorflow.python.keras.activations import relu, gelu, elu, selu

from conf import Conf


class Activation(Conf):

    def __init__(self):
        super(Activation, self).__init__()
        self.activation = self.get_activation()

    def get_activation(self):
        if self.activation_function == 'relu':
            return relu
        elif self.activation_function == 'gelu':
            return gelu
        elif self.activation_function == 'elu':
            return elu
        elif self.activation_function == 'selu':
            return selu
